import torch
A=torch.Tensor([[1,2,3],[1,2,3]])
print(A)
A = A / A.norm(2, 1, keepdim=True).expand_as(A)
print(A)
